package com.z.module.bpm.service.definition;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.bean.copier.CopyOptions;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
import com.z.module.bpm.domain.definition.BpmTaskAssignRuleDO;
import com.z.module.bpm.domain.definition.BpmUserGroupDO;
import com.z.module.bpm.framework.flowable.core.behavior.script.BpmTaskAssignScript;
import com.z.module.bpm.repository.definition.BpmTaskAssignRuleRepository;
import com.z.module.bpm.util.FlowableUtils;
import com.z.module.bpm.web.mapper.definition.BpmTaskAssignRuleConvert;
import com.z.module.bpm.web.vo.definition.rule.BpmTaskAssignRuleCreateReqVO;
import com.z.module.bpm.web.vo.definition.rule.BpmTaskAssignRuleRespVO;
import com.z.module.bpm.web.vo.definition.rule.BpmTaskAssignRuleUpdateReqVO;
import lombok.experimental.Accessors;
import lombok.extern.slf4j.Slf4j;
import org.flowable.bpmn.model.BpmnModel;
import org.flowable.bpmn.model.UserTask;
import org.flowable.common.engine.api.FlowableException;
import org.flowable.engine.delegate.DelegateExecution;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;

import jakarta.annotation.Resource;
import javax.validation.Valid;
import java.util.*;

import static cn.hutool.core.text.CharSequenceUtil.format;
import static com.z.framework.common.exception.util.ServiceExceptionUtil.exception;
import static com.z.framework.common.util.collection.CollectionUtils.convertMap;
import static com.z.module.bpm.enums.ErrorCodeConstants.*;

/**
 * BPM 任务分配规则 Service 实现类
 */
@Service
@Validated
@Slf4j
@Accessors(chain = true)
public class BpmTaskAssignRuleServiceImpl implements BpmTaskAssignRuleService {

    @Autowired
    private BpmTaskAssignRuleRepository bpmTaskAssignRuleRepository;

    @Autowired
    @Lazy // 解决循环依赖
    private BpmModelService modelService;

    @Autowired
    @Lazy // 解决循环依赖
    private BpmProcessDefinitionService processDefinitionService;

    @Autowired
    private BpmUserGroupService userGroupService;

    @Autowired
    private BpmTaskAssignRuleConvert bpmTaskAssignRuleConvert;

    /**
     * 任务分配脚本
     */
    private Map<Long, BpmTaskAssignScript> scriptMap = Collections.emptyMap();

    @Resource
    public void setScripts(List<BpmTaskAssignScript> scripts) {
        this.scriptMap = convertMap(scripts, script -> script.getEnum().getId());
    }

    @Override
    public List<BpmTaskAssignRuleDO> getTaskAssignRuleListByProcessDefinitionId(String processDefinitionId,
                                                                                String taskDefinitionKey) {
        return bpmTaskAssignRuleRepository.findAllByProcessDefinitionIdAndTaskDefinitionKey(processDefinitionId, taskDefinitionKey);
    }

    @Override
    public List<BpmTaskAssignRuleDO> getTaskAssignRuleListByModelId(String modelId) {
        return bpmTaskAssignRuleRepository.findAllByModelId(modelId);
    }

    @Override
    public List<BpmTaskAssignRuleRespVO> getTaskAssignRuleList(String modelId, String processDefinitionId) {
        // 获得规则
        List<BpmTaskAssignRuleDO> rules = Collections.emptyList();
        BpmnModel model = null;
        if (StrUtil.isNotEmpty(modelId)) {
            rules = getTaskAssignRuleListByModelId(modelId);
            model = modelService.getBpmnModel(modelId);
        } else if (StrUtil.isNotEmpty(processDefinitionId)) {
            rules = getTaskAssignRuleListByProcessDefinitionId(processDefinitionId, null);
            model = processDefinitionService.getBpmnModel(processDefinitionId);
        }
        if (model == null) {
            return Collections.emptyList();
        }
        // 获得用户任务，只有用户任务才可以设置分配规则
        List<UserTask> userTasks = FlowableUtils.getBpmnModelElements(model, UserTask.class);
        if (CollUtil.isEmpty(userTasks)) {
            return Collections.emptyList();
        }
        // 转换数据
        return bpmTaskAssignRuleConvert.convertList(userTasks, rules);
    }

    @Override
    public Long createTaskAssignRule(@Valid BpmTaskAssignRuleCreateReqVO reqVO) {
        // 校验参数
//        validTaskAssignRuleOptions(reqVO.getType(), reqVO.getOptions());
        // 校验是否已经配置
        final List<BpmTaskAssignRuleDO> allByModelIdAndTaskDefinitionKey = bpmTaskAssignRuleRepository.findAllByModelIdAndTaskDefinitionKey(reqVO.getModelId(), reqVO.getTaskDefinitionKey());
        if (allByModelIdAndTaskDefinitionKey.size() > 0) {
            throw exception(TASK_ASSIGN_RULE_EXISTS, reqVO.getModelId(), reqVO.getTaskDefinitionKey());
        }

        // 存储
        BpmTaskAssignRuleDO rule = bpmTaskAssignRuleConvert.convert(reqVO)
            .setProcessDefinitionId(BpmTaskAssignRuleDO.PROCESS_DEFINITION_ID_NULL); // 只有流程模型，才允许新建
        bpmTaskAssignRuleRepository.save(rule);
        return rule.getId();
    }

    @Override
    public void updateTaskAssignRule(@Valid BpmTaskAssignRuleUpdateReqVO reqVO) {
        // 校验参数
//        validTaskAssignRuleOptions(reqVO.getType(), reqVO.getOptions());
        // 校验是否存在
        Optional<BpmTaskAssignRuleDO> existRuleOptional = bpmTaskAssignRuleRepository.findById(reqVO.getId());
        if (!existRuleOptional.isPresent()) {
            throw exception(TASK_ASSIGN_RULE_NOT_EXISTS);
        }
        // 只允许修改流程模型的规则
        if (!Objects.equals(BpmTaskAssignRuleDO.PROCESS_DEFINITION_ID_NULL, existRuleOptional.get().getProcessDefinitionId())) {
            throw exception(TASK_UPDATE_FAIL_NOT_MODEL);
        }

        // 执行更新
        final BpmTaskAssignRuleDO bpmTaskAssignRuleDO = existRuleOptional.get();
        final BpmTaskAssignRuleDO convert = bpmTaskAssignRuleConvert.convert(reqVO);
        final CopyOptions copyOptions = CopyOptions.create();
        copyOptions.setIgnoreNullValue(true);
        BeanUtil.copyProperties(convert, bpmTaskAssignRuleDO, copyOptions);
        bpmTaskAssignRuleRepository.save(bpmTaskAssignRuleDO);
    }

    @Override
    public boolean isTaskAssignRulesEquals(String modelId, String processDefinitionId) {
        // 调用 VO 接口的原因是，过滤掉流程模型不需要的规则，保持和 copyTaskAssignRules 方法的一致性
        List<BpmTaskAssignRuleRespVO> modelRules = getTaskAssignRuleList(modelId, null);
        List<BpmTaskAssignRuleRespVO> processInstanceRules = getTaskAssignRuleList(null, processDefinitionId);
        if (modelRules.size() != processInstanceRules.size()) {
            return false;
        }

        // 遍历，匹配对应的规则
        Map<String, BpmTaskAssignRuleRespVO> processInstanceRuleMap =
            convertMap(processInstanceRules, BpmTaskAssignRuleRespVO::getTaskDefinitionKey);
        for (BpmTaskAssignRuleRespVO modelRule : modelRules) {
            BpmTaskAssignRuleRespVO processInstanceRule = processInstanceRuleMap.get(modelRule.getTaskDefinitionKey());
            if (processInstanceRule == null) {
                return false;
            }
            if (!ObjectUtil.equals(modelRule.getType(), processInstanceRule.getType()) || !ObjectUtil.equal(
                modelRule.getOptions(), processInstanceRule.getOptions())) {
                return false;
            }
        }
        return true;
    }

    @Override
    public void copyTaskAssignRules(String fromModelId, String toProcessDefinitionId) {
        List<BpmTaskAssignRuleRespVO> rules = getTaskAssignRuleList(fromModelId, null);
        if (CollUtil.isEmpty(rules)) {
            return;
        }
        // 开始复制
        List<BpmTaskAssignRuleDO> newRules = BpmTaskAssignRuleConvert.INSTANCE.convertList2(rules);
        newRules.forEach(rule -> rule.setProcessDefinitionId(toProcessDefinitionId).setId(null));
        bpmTaskAssignRuleRepository.saveAll(newRules);
    }

    @Override
    public void checkTaskAssignRuleAllConfig(String id) {
        // 一个用户任务都没配置，所以无需配置规则
        List<BpmTaskAssignRuleRespVO> taskAssignRules = getTaskAssignRuleList(id, null);
        if (CollUtil.isEmpty(taskAssignRules)) {
            return;
        }
        // 校验未配置规则的任务
        taskAssignRules.forEach(rule -> {
            if (CollUtil.isEmpty(rule.getOptions())) {
                throw exception(MODEL_DEPLOY_FAIL_TASK_ASSIGN_RULE_NOT_CONFIG, rule.getTaskDefinitionName());
            }
        });
    }

//    private void validTaskAssignRuleOptions(Integer type, Set<Long> options) {
//        if (Objects.equals(type, BpmTaskAssignRuleTypeEnum.ROLE.getType())) {
//            roleApi.validRoleList(options);
//        } else if (ObjectUtils.equalsAny(type, BpmTaskAssignRuleTypeEnum.DEPT_MEMBER.getType(),
//            BpmTaskAssignRuleTypeEnum.DEPT_LEADER.getType())) {
//            deptApi.validateDeptList(options);
//        } else if (Objects.equals(type, BpmTaskAssignRuleTypeEnum.POST.getType())) {
//            postApi.validPostList(options);
//        } else if (Objects.equals(type, BpmTaskAssignRuleTypeEnum.USER.getType())) {
//            adminUserApi.validateUserList(options);
//        } else if (Objects.equals(type, BpmTaskAssignRuleTypeEnum.USER_GROUP.getType())) {
//            userGroupService.validUserGroups(options);
//        } else if (Objects.equals(type, BpmTaskAssignRuleTypeEnum.SCRIPT.getType())) {
//            dictDataApi.validateDictDataList(DictTypeConstants.TASK_ASSIGN_SCRIPT,
//                convertSet(options, String::valueOf));
//        } else {
//            throw new IllegalArgumentException(format("未知的规则类型({})", type));
//        }
//    }

    @Override
    public Set<Long> calculateTaskCandidateUsers(DelegateExecution execution) {
        BpmTaskAssignRuleDO rule = getTaskRule(execution);
        return calculateTaskCandidateUsers(execution, rule);
    }

    BpmTaskAssignRuleDO getTaskRule(DelegateExecution execution) {
        List<BpmTaskAssignRuleDO> taskRules = getTaskAssignRuleListByProcessDefinitionId(
                execution.getProcessDefinitionId(), execution.getCurrentActivityId());
        if (CollUtil.isEmpty(taskRules)) {
            throw new FlowableException(format("流程任务({}/{}/{}) 找不到符合的任务规则",
                    execution.getId(), execution.getProcessDefinitionId(), execution.getCurrentActivityId()));
        }
        if (taskRules.size() > 1) {
            throw new FlowableException(format("流程任务({}/{}/{}) 找到过多任务规则({})",
                    execution.getId(), execution.getProcessDefinitionId(), execution.getCurrentActivityId()));
        }
        return taskRules.get(0);
    }

    Set<Long> calculateTaskCandidateUsers(DelegateExecution execution, BpmTaskAssignRuleDO rule) {
        Set<Long> assigneeUserIds = null;
//        if (Objects.equals(BpmTaskAssignRuleTypeEnum.ROLE.getType(), rule.getType())) {
//            assigneeUserIds = calculateTaskCandidateUsersByRole(rule);
//        } else if (Objects.equals(BpmTaskAssignRuleTypeEnum.DEPT_MEMBER.getType(), rule.getType())) {
//            assigneeUserIds = calculateTaskCandidateUsersByDeptMember(rule);
//        } else if (Objects.equals(BpmTaskAssignRuleTypeEnum.DEPT_LEADER.getType(), rule.getType())) {
//            assigneeUserIds = calculateTaskCandidateUsersByDeptLeader(rule);
//        } else if (Objects.equals(BpmTaskAssignRuleTypeEnum.POST.getType(), rule.getType())) {
//            assigneeUserIds = calculateTaskCandidateUsersByPost(rule);
//        } else if (Objects.equals(BpmTaskAssignRuleTypeEnum.USER.getType(), rule.getType())) {
//            assigneeUserIds = calculateTaskCandidateUsersByUser(rule);
//        } else if (Objects.equals(BpmTaskAssignRuleTypeEnum.USER_GROUP.getType(), rule.getType())) {
//            assigneeUserIds = calculateTaskCandidateUsersByUserGroup(rule);
//        } else if (Objects.equals(BpmTaskAssignRuleTypeEnum.SCRIPT.getType(), rule.getType())) {
//            assigneeUserIds = calculateTaskCandidateUsersByScript(execution, rule);
//        }

        // 测试 临时加入候选人
        assigneeUserIds.add(1L);
        assigneeUserIds.add(2L);
        assigneeUserIds.add(3L);

        // 移除被禁用的用户
//        removeDisableUsers(assigneeUserIds);
        // 如果候选人为空，抛出异常
        if (CollUtil.isEmpty(assigneeUserIds)) {
            log.error("[calculateTaskCandidateUsers][流程任务({}/{}/{}) 任务规则({}) 找不到候选人]", execution.getId(),
                    execution.getProcessDefinitionId(), execution.getCurrentActivityId(), JSON.toJSONString(rule));
            throw exception(TASK_CREATE_FAIL_NO_CANDIDATE_USER);
        }
        return assigneeUserIds;
    }

//    private Set<Long> calculateTaskCandidateUsersByRole(BpmTaskAssignRuleDO rule) {
//        return permissionApi.getUserRoleIdListByRoleIds(rule.getOptions());
//    }
//
//    private Set<Long> calculateTaskCandidateUsersByDeptMember(BpmTaskAssignRuleDO rule) {
//        List<AdminUserRespDTO> users = adminUserApi.getUserListByDeptIds(rule.getOptions());
//        return convertSet(users, AdminUserRespDTO::getId);
//    }
//
//    private Set<Long> calculateTaskCandidateUsersByDeptLeader(BpmTaskAssignRuleDO rule) {
//        List<DeptRespDTO> depts = deptApi.getDeptList(rule.getOptions());
//        return convertSet(depts, DeptRespDTO::getLeaderUserId);
//    }
//
//    private Set<Long> calculateTaskCandidateUsersByPost(BpmTaskAssignRuleDO rule) {
//        List<AdminUserRespDTO> users = adminUserApi.getUsersByPostIds(rule.getOptions());
//        return convertSet(users, AdminUserRespDTO::getId);
//    }

    private Set<Long> calculateTaskCandidateUsersByUser(BpmTaskAssignRuleDO rule) {
        return rule.getOptions();
    }

    private Set<Long> calculateTaskCandidateUsersByUserGroup(BpmTaskAssignRuleDO rule) {
        List<BpmUserGroupDO> userGroups = userGroupService.getUserGroupList(rule.getOptions());
        Set<Long> userIds = new HashSet<>();
        userGroups.forEach(group -> userIds.addAll(group.getMemberUserIds()));
        return userIds;
    }

    private Set<Long> calculateTaskCandidateUsersByScript(DelegateExecution execution, BpmTaskAssignRuleDO rule) {
        // 获得对应的脚本
        List<BpmTaskAssignScript> scripts = new ArrayList<>(rule.getOptions().size());
        rule.getOptions().forEach(id -> {
            BpmTaskAssignScript script = scriptMap.get(id);
            if (script == null) {
                throw exception(TASK_ASSIGN_SCRIPT_NOT_EXISTS, id);
            }
            scripts.add(script);
        });
        // 逐个计算任务
        Set<Long> userIds = new HashSet<>();
        scripts.forEach(script -> CollUtil.addAll(userIds, script.calculateTaskCandidateUsers(execution)));
        return userIds;
    }

//    void removeDisableUsers(Set<Long> assigneeUserIds) {
//        if (CollUtil.isEmpty(assigneeUserIds)) {
//            return;
//        }
//        Map<Long, AdminUserRespDTO> userMap = adminUserApi.getUserMap(assigneeUserIds);
//        assigneeUserIds.removeIf(id -> {
//            AdminUserRespDTO user = userMap.get(id);
//            return user == null || !CommonStatusEnum.ENABLE.getStatus().equals(user.getStatus());
//        });
//    }

}
